import socket
import os
import struct
import ctypes
import time

from NetProtocolHeader import ICMP

HOST = '192.168.135.1'


def main(host_addr='127.0.0.1', socket_protocol=socket.IPPROTO_ICMP):
    sniffer = socket.socket(socket.AF_INET, socket.SOCK_RAW, socket_protocol)
    sniffer.bind((host_addr, 0))
    sniffer.setsockopt(socket.IPPROTO_IP, socket.IP_HDRINCL, 1)

    while 1:
        raw_buffer = sniffer.recvfrom(65565)[0]
        ip_header = raw_buffer[0:20]
        iph = struct.unpack('!BBHHHBBH4s4s', ip_header)

        version_ihl = iph[0]
        version = version_ihl >> 4
        ihl = version_ihl & 0xf
        iph_length = ihl * 4
        ttl = iph[5]
        protocol = iph[6]
        s_addr = socket.inet_ntoa(iph[8])
        d_addr = socket.inet_ntoa(iph[9])

        print("IP->version: " + str(version) + ", Header length:  " +
              str(ihl) + ", TTL: " + str(ttl) + ", Protocol: " +
              str(protocol) + ", Source: " + str(s_addr) + ", Destination: " +
              str(d_addr))

        buf = raw_buffer[iph_length: iph_length+ctypes.sizeof(ICMP)]
        icmp_header = ICMP(buf)
        print("ICMP->Type: %d, Code: %d \t(%s)\n" %
              (icmp_header.type, icmp_header.code, time.asctime()))



if __name__ == "__main__":
    main('192.168.192.214')
